package com.lenovo.lcdm.dcm.common.context;

import cn.hutool.core.util.StrUtil;
import com.lenovo.lcdm.common.exception.InvalidTokenException;
import com.lenovo.lcdm.dcm.common.token.Authorized;
import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

/**
 * ITCode resolver
 *
 * @author
 */
public class ItCodeResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(Authorized.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        String itCode = LcdmUserContext.getUser().getItCode();
        if (StrUtil.isBlank(itCode)) {
            throw new InvalidTokenException();
        }
        return itCode;
    }
}
